#ifndef IMAGECLASSIFFIER_H
#define IMAGECLASSIFFIER_H

#include <QMainWindow>
#include <opencv2/ml.hpp>
#include <opencv2/opencv.hpp>

using namespace cv;
using namespace cv::ml;

QT_BEGIN_NAMESPACE
namespace Ui
{
class ImageClassiffier;
}
QT_END_NAMESPACE

class ImageClassiffier : public QMainWindow
{
    Q_OBJECT

public:
    ImageClassiffier( QWidget* parent = nullptr );
    ~ImageClassiffier();

    // 定义一个函数来提取HOG特征
    static Mat extractHOGFeatures( const Mat& image );
    // 训练SVM模型
    static Ptr<SVM> trainSVM( const String& datadir );

private:
    Ui::ImageClassiffier* ui;
    Ptr<SVM>              svm;

private slots:
    void classifyImage();
};
#endif  // IMAGECLASSIFFIER_H
